#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#
# Copyright (C) 2024 reinovo, Inc. All Rights Reserved 
#
# @Time    : 2024/03/14 上午10:24
# @Author  : hmm
# @Email   : liuyuhang0531@foxmail.com
# @File    : detection.py
# @Software: PyCharm

import apriltag
import cv2 as cv

class Detect(object):

    def __init__(self) -> None:
        self.at_detector = apriltag.Detector(apriltag.DetectorOptions(families='tag36h11 tag36h10'))
    
    def tag_detect(self,image):
        gray = cv.cvtColor(image,cv.COLOR_BGR2GRAY)
        tags = self.at_detector.detect(image)
        return tags

    
